#pragma once
// 编译模块
#include <iostream>
#include <string>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "../com/util.hpp"
#include "../com/log.hpp"
namespace lh_compiler
{
    using namespace lh_util;
    using namespace lh_log;
    class Compiler
    {
    public:
        // 编译功能
        static bool Compile(const std::string &file_name)
        {
            pid_t pid = fork();
            if (pid < 0)
            {
                LOG(ERROR) << "compile error, 创建子进程失败" << "\n";
                return false;
            }
            else if (pid == 0)
            {
                // 子进程
                umask(0);
                int err_fd = open(PathUtil::CompileErr(file_name).c_str(), O_CREAT | O_WRONLY, 0664);
                if (err_fd < 0)
                {
                    std::cout << PathUtil::CompileErr(file_name) << std::endl;
                    LOG(ERROR) << "compile error, 创建/打开文件失败" << "\n";
                    exit(1);
                }
                // 重定向
                dup2(err_fd, 2);
                // 程序替换
                // g++ -o xxx.exe xxx.cpp
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),
                       PathUtil::Src(file_name).c_str(), "-std=c++11","-D","COMPILER_ONLINE",nullptr);
                LOG(DEBUG) << "compile error, 调用g++编译器失败" << "\n";
                exit(2);
            }
            else
            {
                // 父进程
                waitpid(pid, nullptr, 0);
                // 判断是否编译成功(编译成功 -> 形成可执行文件)
                if (FileUtil::IsExist(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << "compile success\n";
                    return true;
                }
            }
            LOG(WARNING) << "compile error\n";
            return false;
        }
    };
}